Python dictionary comprehension example